package com.coderly.onlinegallerysharingplatform.controller;

import cn.dev33.satoken.annotation.SaCheckPermission;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.coderly.onlinegallerysharingplatform.common.auth.SaSpaceCheckPermission;
import com.coderly.onlinegallerysharingplatform.common.constant.SpaceUserPermissionConstant;
import com.coderly.onlinegallerysharingplatform.common.convention.errorcode.ErrorCode;
import com.coderly.onlinegallerysharingplatform.common.convention.exception.BizException;
import com.coderly.onlinegallerysharingplatform.common.response.Response;
import com.coderly.onlinegallerysharingplatform.enums.SpaceLevelEnum;
import com.coderly.onlinegallerysharingplatform.enums.SpaceTypeEnum;
import com.coderly.onlinegallerysharingplatform.model.dao.entity.SpaceDO;
import com.coderly.onlinegallerysharingplatform.model.dto.req.space.*;
import com.coderly.onlinegallerysharingplatform.model.dto.resp.space.GetSpaceLevelRespDTO;
import com.coderly.onlinegallerysharingplatform.model.dto.resp.space.SpaceRespDTO;
import com.coderly.onlinegallerysharingplatform.service.SpaceService;
import com.coderly.onlinegallerysharingplatform.toolkit.ExceptionTrowUtils;
import com.coderly.onlinegallerysharingplatform.toolkit.StpKit;
import com.coderly.onlinegallerysharingplatform.toolkit.UserUtil;
import org.springframework.beans.BeanUtils;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

@RestController
public class SpaceController {

    @Resource
    private SpaceService spaceService;

    /**
     * 新建私有空间
     *
     * @param createSpaceReqDTO
     * @return
     */
    @PostMapping("/space/create")
    public Response<Long> createSpace(@RequestBody CreateSpaceReqDTO createSpaceReqDTO) {
        ExceptionTrowUtils.throwIf(createSpaceReqDTO == null, ErrorCode.PARAMS_ERROR);
        Long spaceId = spaceService.createSpace(createSpaceReqDTO);
        return Response.success(spaceId);
    }

    /**
     * 获取空间级别列表
     *
     * @return
     */
    @GetMapping("/space/getSpaceLevel")
    public Response<List<GetSpaceLevelRespDTO>> getSpaceLevel() {
        List<GetSpaceLevelRespDTO> spaceLevelList = Arrays.stream(SpaceLevelEnum.values())
                .map(spaceLevelEnum -> new GetSpaceLevelRespDTO(
                        spaceLevelEnum.getValue(),
                        spaceLevelEnum.getText(),
                        spaceLevelEnum.getMaxCount(),
                        spaceLevelEnum.getMaxSize()
                ))
                .collect(Collectors.toList());
        return Response.success(spaceLevelList);

    }


    /**
     * 更新空间（仅管理员可用）
     */
    @PostMapping("/admin/space/update")
    public Response<Boolean> updateSpace(@RequestBody UpdateSpaceReqDTO updateSpaceReqDTO) {
        if (updateSpaceReqDTO == null || updateSpaceReqDTO.getId() <= 0) {
            throw new BizException(ErrorCode.PARAMS_ERROR);
        }
        // 将实体类和 DTO 进行转换
        SpaceDO spaceDO = new SpaceDO();
        BeanUtils.copyProperties(updateSpaceReqDTO, spaceDO);
        // 自动填充数据
        spaceService.fillSpaceDefaultFieldBySpaceLevel(spaceDO);
        // 数据校验
        spaceService.validSpace(spaceDO, false);
        // 判断是否存在
        long id = updateSpaceReqDTO.getId();
        SpaceDO oldSpace = spaceService.getById(id);
        ExceptionTrowUtils.throwIf(oldSpace == null, ErrorCode.NOT_FOUND_ERROR);
        // 操作数据库
        boolean result = spaceService.updateById(spaceDO);
        ExceptionTrowUtils.throwIf(!result, ErrorCode.OPERATION_ERROR);
        return Response.success(true);
    }

    /**
     * 删除空间 (用户和管理员都可以用)
     */
    @PostMapping("/space/delete")
    public Response<Boolean> deleteSpace(@RequestBody DeleteSpaceReqDTO deleteSpaceReqDTO) {
        if (deleteSpaceReqDTO == null || deleteSpaceReqDTO.getId() <= 0) {
            throw new BizException(ErrorCode.PARAMS_ERROR);
        }
        return spaceService.deleteSpace(deleteSpaceReqDTO);
    }

    /**
     * 编辑空间（用户和管理员都可以使用）
     */
    @PostMapping("/space/edit")
    public Response<Boolean> editSpace(@RequestBody EditSpaceReqDTO editSpaceReqDTO) {
        if (editSpaceReqDTO == null || editSpaceReqDTO.getId() <= 0) {
            throw new BizException(ErrorCode.PARAMS_ERROR);
        }
        return spaceService.editSpace(editSpaceReqDTO);
    }

    /**
     * 根据 id 获取空间（仅管理员可用）
     */
    @GetMapping("/admin/space/get")
    public Response<SpaceDO> getSpaceById(long id) {
        ExceptionTrowUtils.throwIf(id <= 0, ErrorCode.PARAMS_ERROR);
        // 查询数据库
        SpaceDO spaceDO = spaceService.getById(id);
        ExceptionTrowUtils.throwIf(spaceDO == null, ErrorCode.NOT_FOUND_ERROR);
        // 获取封装类
        return Response.success(spaceDO);
    }

    /**
     * 根据 id 获取空间（封装类）
     */
    @GetMapping("/space/get/vo")
    // @SaCheckPermission(type = StpKit.SPACE_TYPE, value = {SpaceUserPermissionConstant.PICTURE_VIEW})
    //@SaSpaceCheckPermission(value = SpaceUserPermissionConstant.SPACE_USER_MANAGE)
    public Response<SpaceRespDTO> getSpaceVOById(Long id) {
        ExceptionTrowUtils.throwIf(id <= 0, ErrorCode.PARAMS_ERROR);
        LambdaQueryWrapper<SpaceDO> eq = Wrappers.lambdaQuery(SpaceDO.class)
                .eq(SpaceDO::getId, id);
        SpaceDO spaceDO = spaceService.getOne(eq);
        ExceptionTrowUtils.throwIf(spaceDO == null, ErrorCode.NOT_FOUND_ERROR);
        // 校验权限
        Integer spaceType = spaceDO.getSpaceType();
        if (SpaceTypeEnum.PRIVATE.getValue() == spaceType) {
            // 私有空间，只有该空间属于当前登录用户，才可以获取空间详情
            Long loginUserID = UserUtil.getLoginUserId();
            if (!loginUserID.equals(spaceDO.getUserId())) {
                throw new BizException(ErrorCode.NO_AUTH_ERROR, "无权限获取空间详情");
            }
        }
        if (SpaceTypeEnum.TEAM.getValue() == spaceType) {
            // 团队空间，只有具有具有指定权限，才可以获取空间详情
            boolean hasPermission = StpKit.SPACE.hasPermission(SpaceUserPermissionConstant.TEAM_SPACE_GET_INFO);
            ExceptionTrowUtils.throwIf(!hasPermission, ErrorCode.NO_AUTH_ERROR, "无权限获取空间详情");
        }
        return Response.success(spaceService.getSpaceRespDTO(spaceDO));
    }

    /**
     * 分页获取图片列表（仅管理员可用）
     */
    @PostMapping("/admin/space/list/page")
    public Response<Page<SpaceDO>> listSpaceByPage(@RequestBody QuerySpaceReqDTO querySpaceReqDTO) {
        long current = querySpaceReqDTO.getCurrent();
        long size = querySpaceReqDTO.getPageSize();
        // 查询数据库
        Page<SpaceDO> spaceDOPage = spaceService.page(new Page<>(current, size),
                spaceService.buildSpaceQueryWrapper(querySpaceReqDTO));
        return Response.success(spaceDOPage);
    }

    @PostMapping("/space/list/page/vo")
    public Response<Page<SpaceRespDTO>> listSpaceVOByPage(@RequestBody QuerySpaceReqDTO querySpaceReqDTO) {
        long current = querySpaceReqDTO.getCurrent();
        long size = querySpaceReqDTO.getPageSize();
        // 限制爬虫
        ExceptionTrowUtils.throwIf(size > 20, ErrorCode.PARAMS_ERROR);
        // 查询数据库
        Page<SpaceDO> spaceDOPage = spaceService.page(new Page<>(current, size),
                spaceService.buildSpaceQueryWrapper(querySpaceReqDTO));
        // 获取封装类
        return Response.success(spaceService.getSpaceRespDTOPage(spaceDOPage));

    }
}
